import os
from openai import OpenAI
# https://help.aliyun.com/zh/model-studio/deepseek-api?spm=a2c4g.11186623.help-menu-2400256.d_2_1_1.71114823t1z7nn
def get_response(messages):
    client = OpenAI(
        # 若没有配置环境变量，请用百炼API Key将下行替换为：api_key="sk-xxx",
        api_key=os.getenv("DASHSCOPE_API_KEY"),
        base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
    )
    # 模型列表：https://help.aliyun.com/zh/model-studio/getting-started/models
    completion = client.chat.completions.create(model="qwen-plus", messages=messages)
    return completion

# 通过 messages 数组实现上下文管理
messages = [
    {'role': 'user', 'content': '你是一个基金经理'}
]


print("="*20+"第一轮对话"+"="*20)
# 通过reasoning_content字段打印思考过程
# 通过content字段打印最终答案
print("="*20+"最终答案"+"="*20)
print(get_response(messages).choices[0].message.content)

messages.append({'role': 'assistant', 'content': get_response(messages).choices[0].message.content})
messages.append({'role': 'user', 'content': '你是谁'})
print("="*20+"第二轮对话"+"="*20)

# 通过reasoning_content字段打印思考过程
# 通过content字段打印最终答案
print("="*20+"最终答案"+"="*20)
print(get_response(messages).choices[0].message.content)